#!/usr/bin/python3
# -*- coding:utf-8 -*-
import numpy as np
import torch as t

def extract_data():
    # 从二进制文件中读取数据，假设存储的是float16类型的数据
    input_data = np.fromfile("../input/input_x.bin", dtype=np.float16)

    # 假设输入数据的形状为 [127, 127, 127]
    input_data = input_data.reshape(127)  # 调整数据形状，确保它是三维的

    # 从二进制文件中读取数据，假设存储的是float16类型的数据
    result_data = np.fromfile("../output/output.bin", dtype=np.float16)

    extracted_data = input_data  # 提取最后一行数据

    print(f"正确的结果应该是: {extracted_data}")

    # 假设输入数据的形状为 [127, 127, 127]
    result_data = result_data.reshape(127)  # 调整数据形状，确保它是三维的

    print(f"我的算子的结果应该是: {result_data}")


if __name__ == "__main__":
    extract_data()
